package net.zf.context;

import java.util.HashMap;
import java.util.Map;

public class ModelContext {

    public static Map<String, ModelInfo> modelInfoMap = new HashMap<>() {{
        put("11", new ModelInfo("https://dashscope.aliyuncs.com/compatible-mode","sk-eb334170c11940c880c63cab5944eb59","qwen-plus","/v1/chat/completions","/v1/embeddings"));
        put("12", new ModelInfo("https://dashscope.aliyuncs.com/compatible-mode","sk-eb334170c11940c880c63cab5944eb59","deepseek-r1","/v1/chat/completions","/v1/embeddings"));
        put("13", new ModelInfo("https://ark.cn-beijing.volces.com/api","2209e2c0-6fce-442a-bf94-551674865986","kimi-k2-250711","/v3/chat/completions","/v3/embeddings"));
        put("14", new ModelInfo("https://api.siliconflow.cn", "sk-qymoujqjnkaddsawvojoopbjfhlwkexxyzklqiwwxdhldqib","THUDM/GLM-4.1V-9B-Thinking","/v1/chat/completions","/v1/embeddings"));
        put("15", new ModelInfo("https://api.siliconflow.cn", "sk-qymoujqjnkaddsawvojoopbjfhlwkexxyzklqiwwxdhldqib","deepseek-ai/DeepSeek-R1-0528-Qwen3-8B","/v1/chat/completions","/v1/embeddings"));
        put("16", new ModelInfo("https://api.siliconflow.cn", "sk-qymoujqjnkaddsawvojoopbjfhlwkexxyzklqiwwxdhldqib","internlm/internlm2_5-7b-chat","/v1/chat/completions","/v1/embeddings"));

    }};

    public static ModelInfo getModel(String modelId){
        return modelInfoMap.get(modelId);
    }

}
